package com.campus.counseling.config;

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;

@Data
@Configuration
@ConfigurationProperties(prefix = "model.training")
public class ModelTrainingConfig {
    
    // 训练参数
    private int batchSize = 32;
    private int numEpochs = 100;
    private double learningRate = 0.001;
    private double validationSplit = 0.2;
    private int kFolds = 5;
    
    // 早停参数
    private boolean enableEarlyStopping = true;
    private int patience = 10;
    private double minDelta = 0.001;
    
    // 模型保存
    private String modelSaveDir = "models";
    private String bestModelPath = "models/best_model.bin";
    private boolean saveCheckpoints = true;
    private int checkpointFrequency = 10;
} 